write a pthon program to